# -*- coding: utf-8 -*-


from pydantic import BaseModel
from datetime import datetime
from datetime import date as date_


class CreateData(BaseModel):
    date: date_
    confirmed: int = 0
    death: int = 0
    recovered: int = 0


class CreateCity(BaseModel):
    country_code: str = None
    country: str = None
    country_population: int = 0
    province: str = None


class ReadData(CreateData):
    id: int
    city_id: int
    create_date: datetime
    update_date: datetime

    class Config:
        orm_mode = True


class ReadCity(CreateCity):
    id: int
    create_date: datetime
    update_date: datetime

    class Config:
        orm_mode = True

